import lib.baichuan as llm
from jinja2 import Environment, FileSystemLoader
import json
env = Environment(loader=FileSystemLoader('./prompt/'))

VOTE_TEMPLATE = env.get_template('vote.txt')

tar_desc = "喜欢与其他人交流，对八卦信息非常热心，经常在微博上发表自己的观点"

tar_opinion = "我们应该保持理性和客观的态度对待小红的恋情传闻，既要关心她的幸福也要尊重她的隐私。同时，我会继续持谨慎态度，不会仅凭微博就完全相信小红的恋情传闻，而是等待更多的证据或官方消息。"

prompt_content = VOTE_TEMPLATE.render(
    desc=tar_desc,
    opinion=tar_opinion
)
print(prompt_content)
vote_content = llm.send_message(prompt=[{'role':'user', 'content':prompt_content}],json_flag=False)

tmp_dict = {
    'content':vote_content
}
with open('test.json', 'w') as f:
    json.dump(tmp_dict, f,ensure_ascii=False)